#pragma once

#include <iostream>
#include <string>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <atomic>
#include <fstream>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp_files/";

    class TimeUtil
    {
    public:
        //  time.tv_sec :  秒
        //  time.tv_usec : 微秒

        static std::string GetTimeStamp()
        {
            struct timeval time;
            gettimeofday(&time, nullptr);
            return std::to_string(time.tv_sec);
        }

        static std::string GetTimeMS()
        {
            struct timeval time;
            gettimeofday(&time, nullptr);
            return std::to_string(time.tv_sec * 1000 + time.tv_usec / 1000); //  毫秒
        }
    };

    class PathUtil
    {
    public:
        //  生成特定后缀的文件路径
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path + file_name + suffix;
            return path_name;
        }

        //  编译时需要有的临时文件

        //  源文件
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }

        //  可执行程序
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        //  编译错误文件
        static std::string CompilerErr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compiler_error");
        }

        //  运行时需要有的临时文件

        //  标准输入
        static std::string StdIn(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        //  标准输出
        static std::string StdOut(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        //  标准错误
        static std::string StdErr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }

    private:
    };

    class FileUtil
    {
    public:
        //  检查指定文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
                return true;

            return false;
        }

        static std::string UniqueFileName()
        {
            //  通过毫秒级时间戳 + 原子性递增唯一值 来确保唯一性
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMS();
            return ms + "_" + std::to_string(id);
        }

        static bool WriteFile(const std::string &file_name, const std::string &code)
        {
            std::ofstream out(file_name);
            if (!out.is_open())
            {
                return false;
            }

            out.write(code.c_str(), code.size());

            out.close();
            return true;
        }

        static std::string ReadFile(const std::string &file_name, bool keep)
        {
            std::ifstream in(file_name);
            if (!in.is_open())
            {
                return std::string();
            }

            std::string buffer;
            std::string line;
            while (std::getline(in, line))
            {
                buffer += line;
                buffer += (keep ? "\n" : "");
            }

            in.close();
            return buffer;
        }

         
    };

    class StringUtil
    {
    public:
        //  str: 需要切割的字符串
        //  vector<std::string> *target : 输出型参数 存储分割结果的容器
        //  sep: 指定的分隔符
        static void SplitString(const std::string &str, std::vector<std::string> *target, std::string sep)
        {
            //  boost::split: Boost 库中的函数，用于切割字符串
            //  boost::is_any_of(sep): 表示以sep作为分隔符
            //  boost::algorithm::token_compress_on : 一个标志，表示压缩相邻的分隔符（即如果分隔符连续出现，则只保留一个分隔符）
            //  boost::algorithm::token_compress_off :  与on相反
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}